# Databricks notebook source
# MAGIC %md
# MAGIC ## Installations

# COMMAND ----------

# MAGIC %pip install -U mostlyai[local] 
# MAGIC %pip install ipywidgets
# MAGIC %restart_python

# COMMAND ----------

import pandas as pd
from mostlyai.sdk import MostlyAI

mostly = MostlyAI(local=True)

# COMMAND ----------

# MAGIC %md
# MAGIC # Get synthetic data

# COMMAND ----------

df_samples = mostly.synthetic_datasets.get("ce967a36-b232-4871-abc6-a9c536928567")

# COMMAND ----------

df_samples.data()

# COMMAND ----------

# MAGIC %md
# MAGIC # Save synthetic data

# COMMAND ----------

df_documents_personals = spark.createDataFrame(df_samples.data()['documents_personals'])
df_personas = spark.createDataFrame(df_samples.data()['personas'])
df_persona_adreca = spark.createDataFrame(df_samples.data()['persona_adreca'])
df_adreca = spark.createDataFrame(df_samples.data()['adreca'])


# COMMAND ----------

df_documents_personals.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_documents_personals_31_03")
df_personas.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_personas_31_03")
df_persona_adreca.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_persona_adreca_31_03")
df_adreca.write.mode("overwrite").saveAsTable("admin_govern_sta_des.tmp_mostlyai.synthetic_adreca_31_03")
